/*
 * Copyright 2013-2019 Xia Jun(3979434@qq.com).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 ***************************************************************************************
 *                                                                                     *
 *                        Website : http://www.farsunset.com                           *
 *                                                                                     *
 ***************************************************************************************
 */
package com.workcell.workdata.im.component.cim.pusher;

import com.workcell.workdata.im.component.redis.SignalRedisTemplate;
import com.workcell.workdata.im.entity.Message;
import com.workcell.workdata.im.model.app.Origin;
import com.workcell.workdata.im.model.dto.BatchMessage;
import com.workcell.workdata.im.service.MessageIndexService;

import javax.annotation.Resource;
import java.util.Collections;
import java.util.List;
import java.util.UUID;

public abstract class BaseMessagePusher {

	@Resource
	protected MessageIndexService messageIndexService;

	@Resource
	protected SignalRedisTemplate signalRedisTemplate;

	/**
	 * 消息发送之前的预处理
	 * @param message
	 */
	protected void preHandle(Message message, Origin origin){}

	/**
	 * 向用户发送消息
	 * @param message
	 */
	public void push(Message message, Origin origin) {
		this.push(message,Collections.singletonList(message.getReceiver()));
	}

	public final void push(Message message,String ignoredChannel) {
		this.push(message,Collections.singletonList(message.getReceiver()),ignoredChannel);
	}

	public final void push(Message message, List<Long> idList) {
		this.push(message,idList,null);
	}

	public final void push(Message message, List<Long> idList,String ignoredChannel) {

		messageIndexService.add(message,idList);

		BatchMessage batchMessage = new BatchMessage();
		batchMessage.setIdList(idList);
		batchMessage.setMessage(message);
		batchMessage.addExcludedChannel(ignoredChannel);

		this.push(batchMessage);
	}

	protected final void push(BatchMessage batchMessage) {

		/*
		 * 通过发送redis广播，到集群中的每台实例，获得当前UID绑定了连接并推送
		 * @see com.farsunset.hoxin.component.message.PushMessageListener
		 */

		batchMessage.setTraceId(UUID.randomUUID().toString());

		signalRedisTemplate.push(batchMessage);
	}



}
